package com.csust.automaticteach.app.rag.transformer;

import org.springframework.ai.chat.model.ChatModel;
import org.springframework.ai.document.Document;
import org.springframework.ai.model.transformer.SummaryMetadataEnricher;
import org.springframework.stereotype.Component;

import java.util.List;

/**
 * @ ClassName MySummaryEnricher
 * @ Description 总结检索增强
 * @ Author 钰玟
 * @ Date 2025/9/26 下午4:37
 * Version 1.0
 **/
@Component
public class MySummaryEnricher {
    private final SummaryMetadataEnricher enricher;

    public MySummaryEnricher(SummaryMetadataEnricher enricher) {
        this.enricher = enricher;
    }
    public List<Document> enrichDocuments(List<Document> documents) {
        return enricher.apply(documents);
    }
}
